import seaborn as sns
import matplotlib.pyplot as plt

# 读取数据
df_corr_one = id.read_csv('/mnt/car_prices_drop_one.csv')

# 取数据中的数值型列
columns = ['condition', 'odometer', 'mmr']

# 根据取到数值型列，创建数据
df_corr_one = df_corr_one[columns]

# 计算相关系数矩阵并保留两位小数
correlation_matrix = df_corr_one.corr().round(2)

# 创建画布，并显示负号
plt.figure(figsize=(10, 8))
plt.rcParams['axes.unicode_minus'] = False

# 设置中文字体样式以及分辨率，分辨率越高清晰度越高
plt.rcParams['font.sans-serif'] = ['WenQuanYi Zen Hei']
plt.rcParams['figure.dpi'] = 300

# 绘制热力图
sns.heatmap(correlation_matrix, annot=True, cmap='coolwarm', annot_kws={'fontsize': 4})

# 添加标题
plt.title('相关热力图')

# 显示图形
plt.show()

print('相关系数矩阵：\n', correlation_matrix)